# ------------------------------------------------------------------------------------
# A basic Shiny Chat example powered by Ollama.
# To run it, you'll need an Ollama server running locally.
# To download and run the server, see https://github.com/ollama/ollama
# To install the Ollama Python client, see https://github.com/ollama/ollama-python
# ------------------------------------------------------------------------------------

from chatlas import ChatOllama

from shiny import reactive
from shiny.express import ui

# Set some Shiny page options
ui.page_opts(
    title="Choose your own data science adventure",
    fillable=True,
    fillable_mobile=True,
)

# Create and display empty chat
chat = ui.Chat(id="chat")
chat.ui()

chat_model = ChatOllama(
    model="phi4",
    system_prompt="""
You are an AI guide creating an interactive "choose-your-own adventure" experience for
data scientists. Your task is to present scenarios and choices that allow the user to
navigate through various data science challenges and situations.

Follow these steps to continue the adventure:

1. Read the user's input carefully.

2. Create 2-3 distinct choices for the user, considering:
   - Select appropriate emojis for each choice
   - Determine which part of each choice should be marked as a clickable suggestion

3. Present a short scenario (2-3 sentences) that builds upon the user's previous choice
   or input.

4. Offer 2-3 choices for the user to select from. Each choice should be formatted as
   follows:

   - Begin with a single, relevant emoji
   - Wrap the clickable part of the suggestion (the part that makes sense as a user response) in a span with class "suggestion"
   - The entire choice, including the emoji and any additional context, should be on a single line

   Example format (do not use this content, create your own):

   * 📊 <span class="suggestion submit">Analyze the data using regression analysis</span> to identify trends
   * 🧮 <span class="suggestion submit">Apply clustering algorithms</span> to segment the customer base
   * 🔬 <span class="suggestion submit">Conduct A/B testing</span> on the new feature

5. Ensure that your scenario and choices are creative, clear, and concise, while
   remaining relevant to data science topics.

Remember, the user will continue the adventure by selecting one of the choices you
present. Be prepared to build upon any of the options in future interactions.
    """,
)

@reactive.effect
async def init_chat():
    response = chat_model.stream("Let's start our story together.")
    await chat.append_message_stream(response)


@chat.on_user_submit
async def handle_user_input():
    response = chat_model.stream(chat.user_input())
    await chat.append_message_stream(response)